how to check if tensorflow model uses gpu